package sis.ra.evaluation;

import java.io.File;
import java.util.ArrayList;

import sis.ra.utility.Utils;


public class extractEntity {

		
	public static void main(String[] args) {
		String filepathstr="C:\\Users\\Administrator\\Desktop\\dataset\\testing company\\";//Google_wiki.txt";//wiki//Amazon.com_wiki.txt";
		File filepath=new File (filepathstr);
		File[] files=filepath.listFiles();
		ArrayList<String> sentence=new ArrayList<String> ();
		for (int i=0;i<files.length;i++)
		{
			if (!files[i].toString().contains("_wiki.pos")) continue;
			if (!files[i].toString().contains("Amazon.com_wiki.pos")) continue;
			String content=Utils.readFile(files[i].toString());//files[i].toString());
			
		//	System.out.println(content);
			
			String [] lines=content.split("\n");
			for (int j=0;j<lines.length;j++)
			{
				sentence.add(lines[j]);
				if (lines[j].contains(".	Punct-Sent"))
				{
					for (int m=0;m<sentence.size();m++)
					{
						System.out.println(sentence.get(m));
					}
					sentence=new ArrayList<String> ();
				}
			}
//			TreeSet<Integer> sortedSet=new TreeSet<Integer>(pna.keySet());
//			Object [] sortedArray=sortedSet.toArray();
//
//			int count=0;
//			String out="";
//			for (int j=0;j<sortedArray.length;j++)
//			{
//				if (entities.get(sortedArray[j]).getEntityType().equals("PERSON")||
//						entities.get(sortedArray[j]).getEntityType().equals("PLACE_REGION")||
//						entities.get(sortedArray[j]).getEntityType().equals("MEASURE")||
//						entities.get(sortedArray[j]).getEntityType().equals("YEAR")||
//						entities.get(sortedArray[j]).getEntityType().equals("DATE")||
//						entities.get(sortedArray[j]).getEntityType().equals("CURRENCY")||
//						entities.get(sortedArray[j]).getEntityType().equals("PERCENT")||
//						entities.get(sortedArray[j]).getEntityType().equals("PERSON_POS")||
//						entities.get(sortedArray[j]).getEntityType().equals("HOLIDAY")||
//						entities.get(sortedArray[j]).getEntityType().equals("CITY")||
//						entities.get(sortedArray[j]).getEntityType().equals("MONTH")||
//						entities.get(sortedArray[j]).getEntityType().equals("TIME_PERIOD")||
//						entities.get(sortedArray[j]).getEntityType().equals("ORGANIZATION")||
//						entities.get(sortedArray[j]).getEntityType().equals("COUNTRY")||
//						entities.get(sortedArray[j]).getEntityType().equals("ADDRESS")||
//						entities.get(sortedArray[j]).getEntityType().equals("FACILITY")
//						)
//					continue;
//				out=sortedArray[j]+"\t"+pna.get(sortedArray[j]).getEntityType()+"\t"+
//						entities.get(sortedArray[j]).getCanoical();	
//				count=count+1;
//				System.out.println(out);
			}
	}
}
